<?php
/**
 * EXPERIMENTAL PostgreSQL database driver, based on PDO.
 * @author Marcos Sanchez [marcosdsanchez at thinkclear dot ar]
 * @package DatabaseAdapters
 */
class QPostgreSqlPdoDatabase extends QPdoDatabase {
		const Adapter = 'PostgreSQL PDO Database Adapter';
		const PDO_PGSQL_DSN_IDENTIFIER = 'pgsql';

		public function Connect() {
				// Lookup Adapter-Specific Connection Properties
				$strDsn = sprintf("%s:host=%s;dbname=%s;port=%s",QPostgreSqlPdoDatabase::PDO_PGSQL_DSN_IDENTIFIER, $this->Server, $this->Database, $this->Port);

				// Connect to the Database Server
				try {
						$this->objPdo = new PDO($strDsn, $this->Username, $this->Password);
				} catch (PDOException $expPgSql) {
						throw new QPostgreSqlDatabaseException(sprintf("Unable to connect to Database: %s",$expPgSql->getMessage()), -1, null);
				}
				// Update Connected Flag
				$this->blnConnectedFlag = true;
		}

		public function Close() {
			parent::Close();

			// Update Connected Flag
			$this->blnConnectedFlag = false;
		}

		public function SqlVariable($mixData, $blnIncludeEquality = false, $blnReverseEquality = false) {
				// Are we SqlVariabling a BOOLEAN value?
				if (is_bool($mixData)) {
						// Yes
						if ($blnIncludeEquality) {
								// We must include the inequality

								if ($blnReverseEquality) {
										// Do a "Reverse Equality"

										// Check against NULL, True then False
										if (is_null($mixData))
												return 'IS NOT NULL';
										else if ($mixData)
												return "= '0'";
										else
												return "!= '0'";
								} else {
										// Check against NULL, True then False
										if (is_null($mixData))
												return 'IS NULL';
										else if ($mixData)
												return "!= '0'";
										else
												return "= '0'";
								}
						} else {
								// Check against NULL, True then False
								if (is_null($mixData))
										return 'NULL';
								else if ($mixData)
										return "'1'";
								else
										return "'0'";
						}
				}

				// Check for Equality Inclusion
				if ($blnIncludeEquality) {
						if ($blnReverseEquality) {
								if (is_null($mixData))
										$strToReturn = 'IS NOT ';
								else
										$strToReturn = '!= ';
						} else {
								if (is_null($mixData))
										$strToReturn = 'IS ';
								else
										$strToReturn = '= ';
						}
				} else
						$strToReturn = '';

				// Check for NULL Value
				if (is_null($mixData))
						return $strToReturn . 'NULL';

				// Check for NUMERIC Value
				if (is_integer($mixData) || is_float($mixData))
						return $strToReturn . sprintf('%s', $mixData);

				// Check for DATE Value
				if ($mixData instanceof QDateTime) {
						if ($mixData->IsTimeNull())
								return $strToReturn . sprintf("'%s'", $mixData->__toString('YYYY-MM-DD'));
						else
								return $strToReturn . sprintf("'%s'", $mixData->__toString(QDateTime::FormatIso));
				}

				// Assume it's some kind of string value
				return $strToReturn . sprintf("'%s'", addslashes($mixData));
		}

		public function SqlLimitVariablePrefix($strLimitInfo) {
				// PostgreSQL uses Limit by Suffixes (via a LIMIT clause)
				// Prefix is not used, therefore, return null
				return null;
		}

		public function SqlLimitVariableSuffix($strLimitInfo) {
				// Setup limit suffix (if applicable) via a LIMIT clause
				if (strlen($strLimitInfo)) {
						if (strpos($strLimitInfo, ';') !== false)
								throw new Exception('Invalid Semicolon in LIMIT Info');
						if (strpos($strLimitInfo, '`') !== false)
								throw new Exception('Invalid Backtick in LIMIT Info');

						// First figure out if we HAVE an offset
						$strArray = explode(',', $strLimitInfo);

						if (count($strArray) == 2) {
								// Yep -- there's an offset
								return sprintf('LIMIT %s OFFSET %s', $strArray[1], $strArray[0]);
						} else if (count($strArray) == 1) {
								return sprintf('LIMIT %s', $strArray[0]);
						} else {
								throw new QPostgreSqlDatabaseException('Invalid Limit Info: ' . $strLimitInfo, 0, null);
						}
				}

				return null;
		}

		public function SqlSortByVariable($strSortByInfo) {
				// Setup sorting information (if applicable) via a ORDER BY clause
				if (strlen($strSortByInfo)) {
						if (strpos($strSortByInfo, ';') !== false)
								throw new Exception('Invalid Semicolon in ORDER BY Info');
						if (strpos($strSortByInfo, '`') !== false)
								throw new Exception('Invalid Backtick in ORDER BY Info');

						return "ORDER BY $strSortByInfo";
				}

				return null;
		}

		public function GetTables() {
				$objResult = $this->Query("SELECT TABLE_NAME FROM INFORMATION_SCHEMA.TABLES WHERE TABLE_SCHEMA = current_schema() ORDER BY TABLE_NAME ASC");
				$strToReturn = array();
				while ($strRowArray = $objResult->FetchRow())
						array_push($strToReturn, $strRowArray[0]);
				return $strToReturn;
		}

		public function GetFieldsForTable($strTableName) {
				$strTableName = $this->SqlVariable($strTableName);
				$strQuery = sprintf('
				SELECT
					table_name,
					column_name,
					ordinal_position,
					column_default,
					is_nullable,
					data_type,
					character_maximum_length,
					(pg_get_serial_sequence(table_name,column_name) IS NOT NULL) AS is_serial
				FROM
					INFORMATION_SCHEMA.COLUMNS
				WHERE
					table_schema = current_schema()
				AND
					table_name = %s
				ORDER BY ordinal_position
			', $strTableName);

				$objResult = $this->Query($strQuery);

				$objFields = array();

				while ($objRow = $objResult->GetNextRow()) {
						array_push($objFields, new QPostgreSqlPdoDatabaseField($objRow, $this));
				}

				return $objFields;
		}

		public function InsertId($strTableName = null, $strColumnName = null) {
				$strQuery = sprintf('
				SELECT currval(pg_get_serial_sequence(%s, %s))
			', $this->SqlVariable($strTableName), $this->SqlVariable($strColumnName));

				$objResult = $this->Query($strQuery);
				$objRow = $objResult->FetchRow();
				return $objRow[0];
		}


		private function ParseColumnNameArrayFromKeyDefinition($strKeyDefinition) {
				$strKeyDefinition = trim($strKeyDefinition);

				// Get rid of the opening "(" and the closing ")"
				$intPosition = strpos($strKeyDefinition, '(');
				if ($intPosition === false)
						throw new Exception("Invalid Key Definition: $strKeyDefinition");
				$strKeyDefinition = trim(substr($strKeyDefinition, $intPosition + 1));

				$intPosition = strpos($strKeyDefinition, ')');
				if ($intPosition === false)
						throw new Exception("Invalid Key Definition: $strKeyDefinition");
				$strKeyDefinition = trim(substr($strKeyDefinition, 0, $intPosition));
				$strKeyDefinition = str_replace(" ","",$strKeyDefinition);

				// Create the Array
				// TODO: Current method doesn't support key names with commas or parenthesis in them!
				$strToReturn = explode(',', $strKeyDefinition);

				// Take out trailing and leading '"' character in each name (if applicable)
				for ($intIndex = 0; $intIndex < count($strToReturn); $intIndex++) {
						$strColumn = $strToReturn[$intIndex];

						if (substr($strColumn, 0, 1) == '"')
								$strColumn = substr($strColumn, 1, strpos($strColumn, '"', 1) - 1);

						$strToReturn[$intIndex] = $strColumn;
				}

				return $strToReturn;
		}

		public function GetIndexesForTable($strTableName) {
				$objIndexArray = array();

				$objResult = $this->Query(sprintf('
				SELECT
					c2.relname AS indname,
					i.indisprimary,
					i.indisunique,
					pg_catalog.pg_get_indexdef(i.indexrelid) AS inddef
				FROM
					pg_catalog.pg_class c,
					pg_catalog.pg_class c2,
					pg_catalog.pg_index i
				WHERE
					c.relname = %s
				AND
					pg_catalog.pg_table_is_visible(c.oid)
				AND
					c.oid = i.indrelid
				AND
					i.indexrelid = c2.oid
				ORDER BY
					c2.relname
			', $this->SqlVariable($strTableName)));

				while ($objRow = $objResult->GetNextRow()) {
						$strIndexDefinition = $objRow->GetColumn('inddef');
						$strKeyName = $objRow->GetColumn('indname');
						$blnPrimaryKey = $objRow->GetColumn('indisprimary');
						$blnUnique = $objRow->GetColumn('indisunique');
						$strColumnNameArray = $this->ParseColumnNameArrayFromKeyDefinition($strIndexDefinition);

						$objIndex = new QDatabaseIndex($strKeyName, $blnPrimaryKey, $blnUnique, $strColumnNameArray);
						array_push($objIndexArray, $objIndex);
				}

				return $objIndexArray;
		}

		public function GetForeignKeysForTable($strTableName) {
				$objForeignKeyArray = array();

				// Use Query to pull the FKs
				$strQuery = sprintf('
				SELECT
					pc.conname,
					pg_catalog.pg_get_constraintdef(pc.oid, true) AS consrc
				FROM
					pg_catalog.pg_constraint pc
				WHERE
					pc.conrelid =
					(
						SELECT
							oid
						FROM
							pg_catalog.pg_class
						WHERE
							relname=%s
						AND
							relnamespace =
							(
								SELECT
									oid
								FROM
									pg_catalog.pg_namespace
								WHERE
									nspname=current_schema()
							)
					)
				AND
					pc.contype = \'f\'
			', $this->SqlVariable($strTableName));

				$objResult = $this->Query($strQuery);

				while ($objRow = $objResult->GetNextRow()) {
						$strKeyName = $objRow->GetColumn('conname');

						// Remove leading and trailing '"' characters (if applicable)
						if (substr($strKeyName, 0, 1) == '"')
								$strKeyName = substr($strKeyName, 1, strlen($strKeyName) - 2);

						// By the end of the following lines, we will end up with a strTokenArray
						// Index 1: the list of columns that are the foreign key
						// Index 2: the table which this FK references
						// Index 3: the list of columns which this FK references
						$strTokenArray = explode('FOREIGN KEY ', $objRow->GetColumn('consrc'));
						$strTokenArray[1] = explode(' REFERENCES ', $strTokenArray[1]);
						$strTokenArray[2] = $strTokenArray[1][1];
						$strTokenArray[1] = $strTokenArray[1][0];
						$strTokenArray[2] = explode("(", $strTokenArray[2]);
						$strTokenArray[3] = "(".$strTokenArray[2][1];
						$strTokenArray[2] = $strTokenArray[2][0];

						// Remove leading and trailing '"' characters (if applicable)
						if (substr($strTokenArray[2], 0, 1) == '"')
								$strTokenArray[2] = substr($strTokenArray[2], 1, strlen($strTokenArray[2]) - 2);

						$strColumnNameArray = $this->ParseColumnNameArrayFromKeyDefinition($strTokenArray[1]);
						$strReferenceTableName = $strTokenArray[2];
						$strReferenceColumnNameArray = $this->ParseColumnNameArrayFromKeyDefinition($strTokenArray[3]);

						$objForeignKey = new QDatabaseForeignKey(
										$strKeyName,
										$strColumnNameArray,
										$strReferenceTableName,
										$strReferenceColumnNameArray);
						array_push($objForeignKeyArray, $objForeignKey);
				}

				// Return the Array of Foreign Keys
				return $objForeignKeyArray;
		}

		protected function ExecuteQuery($strQuery) {
				// Perform the Query
				$objResult = $this->objPdo->query($strQuery);
				if ($objResult === false)
						throw new QPdoDatabaseException($this->objPdo->errorInfo(), $this->objPdo->errorCode(), $strQuery);

				// Return the Result
				$this->objMostRecentResult = $objResult;
				$objPdoStatementDatabaseResult = new QPostgreSqlPdoDatabaseResult($objResult, $this);
				return $objPdoStatementDatabaseResult;
		}


}
/**
 * QPostgreSqlPdoDatabaseResult
 */
class QPostgreSqlPdoDatabaseResult extends QPdoDatabaseResult {

		public function GetNextRow() {
				$strColumnArray = $this->FetchArray();

				if ($strColumnArray)
						return new QPostgreSqlPdoDatabaseRow($strColumnArray);
				else
						return null;
		}

		public function FetchFields() {
				$objArrayToReturn = array();
				while ($objField = $this->FetchColumn()) {
						array_push($objArrayToReturn, new QPostgreSqlPdoDatabaseField($objField, $this->objDb));
				}
				return $objArrayToReturn;
		}

		public function FetchField() {
				if ($objField = $this->FetchColumn())
						return new QPostgreSqlPdoDatabaseField($objField, $this->objDb);
		}

}
/**
 * QPostgreSqlPdoDatabaseRow
 */
class QPostgreSqlPdoDatabaseRow extends QDatabaseRowBase {
		protected $strColumnArray;

		public function __construct($strColumnArray) {
				$this->strColumnArray = $strColumnArray;
		}

		public function GetColumn($strColumnName, $strColumnType = null) {
				if (array_key_exists($strColumnName, $this->strColumnArray)) {
						if (is_null($this->strColumnArray[$strColumnName]))
								return null;

						switch ($strColumnType) {
								case QDatabaseFieldType::Bit:
										if (!$this->strColumnArray[$strColumnName]) {
												return false;
										} else {
												return ($this->strColumnArray[$strColumnName]) ? true : false;
										}

								case QDatabaseFieldType::Blob:
								case QDatabaseFieldType::Char:
								case QDatabaseFieldType::VarChar:
										return QType::Cast($this->strColumnArray[$strColumnName], QType::String);

								case QDatabaseFieldType::Date:
								case QDatabaseFieldType::DateTime:
								case QDatabaseFieldType::Time:
										return new QDateTime($this->strColumnArray[$strColumnName]);

								case QDatabaseFieldType::Float:
										return QType::Cast($this->strColumnArray[$strColumnName], QType::Float);

								case QDatabaseFieldType::Integer:
										return QType::Cast($this->strColumnArray[$strColumnName], QType::Integer);

								default:
										return $this->strColumnArray[$strColumnName];
						}
				} else
						return null;
		}

		public function ColumnExists($strColumnName) {
				return array_key_exists($strColumnName, $this->strColumnArray);
		}

		public function GetColumnNameArray() {
				return $this->strColumnArray;
		}
}
/**
 * QPostgreSqlPdoDatabaseField
 */
class QPostgreSqlPdoDatabaseField extends QDatabaseFieldBase {
		public function __construct($mixFieldData, $objDb = null) {
				$this->strName = $mixFieldData->GetColumn('column_name');
				$this->strOriginalName = $this->strName;
				$this->strTable = $mixFieldData->GetColumn('table_name');
				$this->strOriginalTable = $this->strTable;
				$this->strDefault = $mixFieldData->GetColumn('column_default');
				$this->intMaxLength = $mixFieldData->GetColumn('character_maximum_length', QDatabaseFieldType::Integer);
				$this->blnNotNull = ($mixFieldData->GetColumn('is_nullable') == "NO") ? true : false;

				// If the first column of the table was created as SERIAL, we assume it's the identity field.
				// Otherwise, no identity field will be set for this table.
				$ordinalPos = $mixFieldData->GetColumn('ordinal_position', QDatabaseFieldType::Integer);
				$isSerial = $mixFieldData->GetColumn('is_serial');
				$this->blnIdentity = ($ordinalPos == 1 && $isSerial) ? true : false;

				// Determine Primary Key
				$objResult = $objDb->Query(sprintf('
				SELECT
					kcu.column_name
				FROM
					information_schema.table_constraints tc,
					information_schema.key_column_usage kcu
				WHERE
					tc.table_name = %s
				AND
					tc.table_schema = current_schema()
				AND
					tc.constraint_type = \'PRIMARY KEY\'
				AND
					kcu.table_name = tc.table_name
				AND
					kcu.table_schema = tc.table_schema
				AND
					kcu.constraint_name = tc.constraint_name
			', $objDb->SqlVariable($this->strTable)));

				while ($objRow = $objResult->GetNextRow()) {
						if ($objRow->GetColumn('column_name') == $this->strName)
								$this->blnPrimaryKey = true;
				}

				if (!$this->blnPrimaryKey)
						$this->blnPrimaryKey = false;

				// UNIQUE
				$objResult = $objDb->Query(sprintf('
				SELECT
					kcu.column_name, (SELECT COUNT(*) FROM information_schema.key_column_usage kcu2 WHERE kcu2.constraint_name=kcu.constraint_name ) as unique_fields
				FROM
					information_schema.table_constraints tc,
					information_schema.key_column_usage kcu
				WHERE
					tc.table_name = %s
				AND
					tc.table_schema = current_schema()
				AND
					tc.constraint_type = \'UNIQUE\'
				AND
					kcu.table_name = tc.table_name
				AND
					kcu.table_schema = tc.table_schema
				AND
					kcu.constraint_name = tc.constraint_name
				GROUP BY
					kcu.constraint_name, kcu.column_name
			', $objDb->SqlVariable($this->strTable)));
				while ($objRow = $objResult->GetNextRow()) {
						if ($objRow->GetColumn('column_name') == $this->strName && $objRow->GetColumn('unique_fields') == 1)
								$this->blnUnique = true;
				}
				if (!$this->blnUnique)
						$this->blnUnique = false;

				// Determine Type
				$this->strType = $mixFieldData->GetColumn('data_type');

				switch ($this->strType) {
						case 'integer':
						case 'smallint':
								$this->strType = QDatabaseFieldType::Integer;
								break;
						case 'money':
						// NOTE: The money type is deprecated in PostgreSQL.
								throw new QPostgreSqlDatabaseException('Unsupported Field Type: money.  Use numeric or decimal instead.', 0,null);
								break;
						case 'bigint':
						case 'decimal':
						case 'numeric':
						case 'real':
						// "BIGINT" must be specified here as a float so that PHP can support it's size
						// http://www.postgresql.org/docs/8.2/static/datatype-numeric.html
								$this->strType = QDatabaseFieldType::Float;
								break;
						case 'bit':
								if ($this->intMaxLength == 1)
										$this->strType = QDatabaseFieldType::Bit;
								else
										throw new QPostgreSqlDatabaseException('Unsupported Field Type: bit with MaxLength > 1', 0, null);
								break;
						case 'boolean':
								$this->strType = QDatabaseFieldType::Bit;
								break;
						case 'character':
								$this->strType = QDatabaseFieldType::Char;
								break;
						case 'character varying':
						case 'double precision':
						// NOTE: PHP does not offer full support of double-precision floats.
						// Value will be set as a VarChar which will guarantee that the precision will be maintained.
						//    However, you will not be able to support full typing control (e.g. you would
						//    not be able to use a QFloatTextBox -- only a regular QTextBox)
								$this->strType = QDatabaseFieldType::VarChar;
								break;
						case 'text':
								$this->strType = QDatabaseFieldType::Blob;
								break;
						case 'timestamp':
						case 'timestamp without time zone':
						// System-generated Timestamp values need to be treated as plain text
								$this->strType = QDatabaseFieldType::VarChar;
								$this->blnTimestamp = true;
								break;
						case 'date':
								$this->strType = QDatabaseFieldType::Date;
								break;
						case 'time':
						case 'time without time zone':
								$this->strType = QDatabaseFieldType::Time;
								break;
						default:
								throw new QPostgreSqlDatabaseException('Unsupported Field Type: ' . $this->strType, 0, null);
				}
		}
}
/**
 * QPostgreSqlPdoDatabaseException
 */
class QPostgreSqlPdoDatabaseException extends QPdoDatabaseException {

}


?>
